//package com.mt.common.core.utils;
//
//import com.alibaba.fastjson.JSON;
//import com.mt.common.core.web.base.BaseEntity;
//import com.mt.common.core.web.base.PageResultDTO;
//import org.elasticsearch.action.admin.indices.create.CreateIndexRequest;
//import org.elasticsearch.action.admin.indices.create.CreateIndexResponse;
//import org.elasticsearch.action.admin.indices.delete.DeleteIndexRequest;
//import org.elasticsearch.action.admin.indices.exists.indices.IndicesExistsRequest;
//import org.elasticsearch.action.admin.indices.exists.indices.IndicesExistsResponse;
//import org.elasticsearch.action.bulk.BulkRequest;
//import org.elasticsearch.action.bulk.BulkResponse;
//import org.elasticsearch.action.delete.DeleteRequest;
//import org.elasticsearch.action.delete.DeleteResponse;
//import org.elasticsearch.action.get.GetRequest;
//import org.elasticsearch.action.get.GetResponse;
//import org.elasticsearch.action.index.IndexRequest;
//import org.elasticsearch.action.index.IndexResponse;
//import org.elasticsearch.action.search.SearchRequest;
//import org.elasticsearch.action.search.SearchResponse;
//import org.elasticsearch.action.support.master.AcknowledgedResponse;
//import org.elasticsearch.action.update.UpdateRequest;
//import org.elasticsearch.action.update.UpdateResponse;
//import org.elasticsearch.client.RequestOptions;
//import org.elasticsearch.client.RestHighLevelClient;
//import org.elasticsearch.client.indices.GetIndexRequest;
//import org.elasticsearch.common.text.Text;
//import org.elasticsearch.common.unit.TimeValue;
//import org.elasticsearch.common.xcontent.XContentType;
//import org.elasticsearch.search.SearchHit;
//import org.elasticsearch.search.builder.SearchSourceBuilder;
//import org.elasticsearch.search.fetch.subphase.highlight.HighlightField;
//import org.slf4j.Logger;
//import org.slf4j.LoggerFactory;
//import org.springframework.beans.factory.annotation.Autowired;
//import org.springframework.stereotype.Component;
//
//import java.io.IOException;
//import java.lang.reflect.Method;
//import java.util.*;
//
///**
// * @ClassName: ESUtil
// * @Description:
// * @Author zhanghong
// * @Date 2022/5/25
// */
//@Component
//public class ESUtil {
//    Logger logger = LoggerFactory.getLogger(ESUtil.class);
//    @Autowired
//    private RestHighLevelClient restHighLevelClient;
//    /**
//     * set方法前缀
//     */
//    public static final String SET_METHOD_PREFIX = "set";
//
//    /**
//     * 默认类型
//     */
//    public static final String DEFAULT_TYPE = "_doc";
//
//    /**
//     * 需要过滤的文档数据
//     */
//    public static final String[] IGNORE_KEY = {"@version","type"};
//
//    /**
//     * 超时时间 1s
//     */
//    public static final TimeValue TIME_VALUE_SECONDS = TimeValue.timeValueSeconds(1);
//
//
//    /**
//     * 返回状态-CREATED
//     */
//    public static final String RESPONSE_STATUS_CREATED = "CREATED";
//
//    /**
//     * 返回状态-OK
//     */
//    public static final String RESPONSE_STATUS_OK = "OK";
//
//    /**
//     * 返回状态-NOT_FOUND
//     */
//    public static final String RESPONSE_STATUS_NOT_FOUND = "NOT_FOUND";
//
//
//
//    //创建索引 类似于数据库表
//    public boolean createIndex(String key) {
//        if (!indexExists(key)) {
//            logger.info("Index is not exits!");
//        }
//        try {
//            CreateIndexRequest request = new CreateIndexRequest(key);
//            CreateIndexResponse response = restHighLevelClient.indices().create(request, RequestOptions.DEFAULT);
//            logger.info("执行建立成功？" + response.isAcknowledged());
//            return response.isAcknowledged();
//        } catch (Exception e) {
//            return false;
//        }
//    }
//
//    //判断索引是否存在
//    public boolean indexExists(String index) {
//        try {
//            GetIndexRequest request = new GetIndexRequest(index);
//            boolean exists = restHighLevelClient.indices().exists(request, RequestOptions.DEFAULT);
//            return exists;
//        } catch (Exception e) {
//            return false;
//        }
//    }
//
//    //删除索引
//    public boolean deleteIndex(String index) {
//        try {
//            DeleteIndexRequest request = new DeleteIndexRequest(index);
//            AcknowledgedResponse response = restHighLevelClient.indices().delete(request, RequestOptions.DEFAULT);
//            logger.info("删除索引：{} {}", index, response.isAcknowledged() ? "成功" : "失败");
//            return response.isAcknowledged();
//        } catch (IOException e) {
//            e.printStackTrace();
//            return false;
//        }
//    }
//    /**
//     * 指定文档是否存在
//     *
//     * @param index 索引
//     * @param id    文档id
//     */
//    public boolean isExists(String index, String id) {
//        GetRequest request = new GetRequest(index,DEFAULT_TYPE,id);
//        try {
//            GetResponse response=restHighLevelClient.get(request,RequestOptions.DEFAULT);
//            return response.isExists();
//        }catch (Exception e){
//            e.printStackTrace();
//            return false;
//        }
//    }
//    /**
//     * 根据id查询文档
//     *
//     * @param index 索引
//     * @param type  类型
//     * @param id    文档id
//     * @param clazz 转换目标Class对象
//     * @return 对象
//     */
//    public <T> T selectDocumentById(String index, String id, Class<T> clazz) {
//        try {
//            GetRequest request = new GetRequest(index, DEFAULT_TYPE, id);
//            GetResponse response = restHighLevelClient.get(request, RequestOptions.DEFAULT);
//            if (response.isExists()) {
//                Map<String, Object> sourceAsMap = response.getSourceAsMap();
//                return dealObject(sourceAsMap, clazz);
//            }
//        } catch (Exception e) {
//            e.printStackTrace();
//        }
//        return null;
//    }
//    /**
//     *（筛选条件）获取数据集合
//     * 如果使用排序则 sourceBuilder.sort("name",SortOrder.DESC)
//     * 如果使用高亮则 :
//     * HighlightBuilder highlightBuilder = new HighlightBuilder();
//     * highlightBuilder.field("");
//     * sourceBuilder.highlighter(highlightBuilder);
//     * @param index         索引
//     * @param sourceBuilder 请求条件
//     * @param clazz         转换目标Class对象
//     */
//    public <T> List<T> selectDocumentList(String index, SearchSourceBuilder sourceBuilder, Class<T> clazz) {
//        try {
//            SearchRequest request = new SearchRequest(index);
//            if (sourceBuilder != null) {
//                // 返回实际命中数
//                sourceBuilder.trackTotalHits(true);
//                request.source(sourceBuilder);
//            }
//            SearchResponse response = restHighLevelClient.search(request, RequestOptions.DEFAULT);
//
//            if (response.getHits() != null) {
//                List<T> list = new ArrayList<>();
//                SearchHit[] hits = response.getHits().getHits();
//                for (SearchHit documentFields : hits) {
//                    Map<String, Object> sourceAsMap = documentFields.getSourceAsMap();
//                    // 高亮结果集特殊处理 -- 高亮信息会显示在highlight标签下  需要将实体类中的字段进行替换
//                    Map<String, Object> map = this.highlightBuilderHandle(sourceAsMap, documentFields);
//                    list.add(dealObject(map, clazz));
//                }
//                return list;
//            }
//        } catch (Exception e) {
//            e.printStackTrace();
//        }
//        return null;
//    }
//
//    /**
//     * 新增/修改文档信息
//     * @param index 索引
//     * @param data  数据
//     */
//    public String insertDocument(String index, BaseEntity data) {
//        try {
//            //String id = UUID.randomUUID().toString().replaceAll("-", "").toUpperCase();
//            IndexRequest request = new IndexRequest(index);
//            request.timeout(TIME_VALUE_SECONDS);
//            request.id(String.valueOf(data.getEid())); // 文档id
//            request.source(JSON.toJSONString(data), XContentType.JSON);
//            IndexResponse response = restHighLevelClient.index(request, RequestOptions.DEFAULT);
//            logger.info("insertDocument response status:{},id:{}", response.status().getStatus(), response.getId());
//            String status = response.status().toString();
//            if (RESPONSE_STATUS_CREATED.equals(status) || RESPONSE_STATUS_OK.equals(status)) {
//                return response.getId();
//            }
//        } catch (Exception e) {
//            e.printStackTrace();
//        }
//        return "";
//    }
//
//    /**
//     * 删除文档信息
//     *
//     * @param index 索引
//     * @param id    文档id
//     */
//    public boolean deleteDocument(String index, String id) {
//        try {
//            DeleteRequest request = new DeleteRequest(index, id);
//            DeleteResponse response = restHighLevelClient.delete(request, RequestOptions.DEFAULT);
//            logger.info("deleteDocument response status:{},id:{}", response.status().getStatus(), response.getId());
//            String status = response.status().toString();
//            if (RESPONSE_STATUS_OK.equals(status)) {
//                return true;
//            }
//        } catch (Exception e) {
//            e.printStackTrace();
//        }
//        return false;
//    }
//
//
//    /**
//     * 更新文档信息
//     *
//     * @param index 索引
//     * @param id    文档id
//     * @param data  数据
//     */
//    public boolean updateDocument(String index,BaseEntity data) {
//
//        try {
//            UpdateRequest request = new UpdateRequest(index, String.valueOf(data.getEid()));
//            request.doc(data, XContentType.JSON);
//            UpdateResponse response = restHighLevelClient.update(request, RequestOptions.DEFAULT);
//            logger.info("updateDocument response status:{},id:{}", response.status().getStatus(), response.getId());
//            String status = response.status().toString();
//            if (RESPONSE_STATUS_OK.equals(status)) {
//                return true;
//            }
//        } catch (Exception e) {
//            e.printStackTrace();
//        }
//        return false;
//    }
//
//
//    /**
//     * 批量操作(新增)
//     * @param index    索引
//     * @param opType   操作类型 PATCH_OP_TYPE_*
//     * @param dataList 数据集 新增修改需要传递
//     * @param timeout  超时时间 单位为秒
//     */
//    public boolean patch(String index, String opType, List<? extends BaseEntity> dataList, long timeout) {
//        try {
//            BulkRequest bulkRequest = new BulkRequest();
//            bulkRequest.timeout(TimeValue.timeValueSeconds(timeout));
//            if (dataList != null && dataList.size() > 0) {
//                if ("insert".equals(opType)) {
//                    for (BaseEntity obj : dataList) {
//                        bulkRequest.add(
//                                new IndexRequest(index).id(String.valueOf(obj.getEid()))
//                                        .source(JSON.toJSONString(obj), XContentType.JSON)
//                        );
//                    }
//                }
//                BulkResponse response = restHighLevelClient.bulk(bulkRequest, RequestOptions.DEFAULT);
//                if (!response.hasFailures()) {
//                    return true;
//                }
//            }
//        } catch (Exception e) {
//            e.printStackTrace();
//        }
//        return false;
//    }
//
//    /**
//     *（筛选条件）获取数据集合分页
//     * 如果使用排序则 sourceBuilder.sort("name",SortOrder.DESC)
//     * 如果使用高亮则 :
//     * HighlightBuilder highlightBuilder = new HighlightBuilder();
//     * highlightBuilder.field("");
//     * sourceBuilder.highlighter(highlightBuilder);
//     * @param index         索引
//     * @param sourceBuilder 请求条件
//     * @param clazz         转换目标Class对象
//     */
//    public <T> PageResultDTO selectDocumentPage(String index,SearchSourceBuilder sourceBuilder, int startPage, int pageSize , Class<T> clazz) {
//        try {
//            SearchRequest request = new SearchRequest(index);
//            if (sourceBuilder != null) {
//                // 返回实际命中数
//                sourceBuilder.from(startPage);
//                sourceBuilder.size(pageSize);
//                sourceBuilder.explain(true);
//                sourceBuilder.trackTotalHits(true);
//                request.source(sourceBuilder);
//            }
//            SearchResponse response = restHighLevelClient.search(request, RequestOptions.DEFAULT);
//            if (response.getHits() != null) {
//                long totalHits = Arrays.stream(response.getHits().getHits()).count();
//                long length = response.getHits().getHits().length;
//
//                List<T> list = new ArrayList<>();
//                SearchHit[] hits = response.getHits().getHits();
//                for (SearchHit documentFields : hits) {
//                    Map<String, Object> sourceAsMap = documentFields.getSourceAsMap();
//                    // 高亮结果集特殊处理 -- 高亮信息会显示在highlight标签下  需要将实体类中的字段进行替换
//                    Map<String, Object> map = this.highlightBuilderHandle(sourceAsMap, documentFields);
//                    list.add(dealObject(map, clazz));
//                }
//                return new PageResultDTO( (Long) totalHits, list);
//            }
//        } catch (Exception e) {
//            e.printStackTrace();
//        }
//        return null;
//    }
//
//    /**
//     * 高亮结果集 特殊处理
//     * @param sourceAsMap
//     * @param documentFields
//     */
//    private Map<String, Object> highlightBuilderHandle(Map<String, Object> sourceAsMap,SearchHit documentFields){
//        // 将高亮的字段替换到sourceAsMap中
//        Map<String, HighlightField> fieldMap = documentFields.getHighlightFields();
//        Set<Map.Entry<String, Object>> entries = sourceAsMap.entrySet();
//        entries.forEach(source -> {
//            if (fieldMap.containsKey(source.getKey())){
//                Text[] fragments = fieldMap.get(source.getKey()).getFragments();
//                if (fragments != null){
//                    for (Text str : fragments) {
//                        source.setValue(str.string());
//                    }
//                }
//            }
//        });
//        return sourceAsMap;
//    }
//
////    /**
////     * 高亮结果集 特殊处理
////     *
////     * @param searchResponse
////     * @param highlightField
////     */
////    private List<Map<String, Object>> setSearchResponse(SearchResponse searchResponse, String highlightField) {
////        List<Map<String, Object>> sourceList = new ArrayList<>();
////        StringBuffer stringBuffer = new StringBuffer();
////
////        for (SearchHit searchHit : searchResponse.getHits().getHits()) {
////            searchHit.getSourceAsMap().put("id", searchHit.getId());
////
////            if (StringUtils.isNotEmpty(highlightField)) {
////
////                System.out.println("遍历 高亮结果集，覆盖 正常结果集" + searchHit.getSourceAsMap());
////                Text[] text = searchHit.getHighlightFields().get(highlightField).getFragments();
////
////                if (text != null) {
////                    for (Text str : text) {
////                        stringBuffer.append(str.string());
////                    }
////                    //遍历 高亮结果集，覆盖 正常结果集
////                    searchHit.getSourceAsMap().put(highlightField, stringBuffer.toString());
////                }
////            }
////            sourceList.add(searchHit.getSourceAsMap());
////        }
////        return sourceList;
////    }
//
//
//    /**
//     * 将文档数据转化为指定对象
//     *
//     * @param sourceAsMap 文档数据
//     * @param clazz       转换目标Class对象
//     * @return
//     */
//    private static <T> T dealObject(Map<String, Object> sourceAsMap, Class<T> clazz) {
//        //try {
//        //    ignoreSource(sourceAsMap);
//        //    Iterator<String> keyIterator = sourceAsMap.keySet().iterator();
//        //    T t = clazz.newInstance();
//        //    while (keyIterator.hasNext()) {
//        //        String key = keyIterator.next();
//        //        String replaceKey = key.replaceFirst(key.substring(0, 1), key.substring(0, 1).toUpperCase());
//        //        Method method = null;
//        //        try {
//        //            method = clazz.getMethod(SET_METHOD_PREFIX + replaceKey, sourceAsMap.get(key).getClass());
//        //        }catch (NoSuchMethodException e) {
//        //            Class<? super T> superclass = clazz.getSuperclass();
//        //            method = superclass.getDeclaredMethod(SET_METHOD_PREFIX + replaceKey, sourceAsMap.get(key).getClass());
//        //            continue;
//        //
//        //        }
//        //        method.invoke(t, sourceAsMap.get(key));
//        //    }
//        //    return t;
//        //} catch (Exception e) {
//        //    e.printStackTrace();
//        //}
//        //return null;
//        return JSON.parseObject(JSON.toJSONString(sourceAsMap), clazz);
//    }
//
//    /**
//     * 剔除指定文档数据,减少不必要的循环
//     *
//     * @param map 文档数据
//     */
//    private static void ignoreSource(Map<String, Object> map) {
//        for (String key : IGNORE_KEY) {
//            map.remove(key);
//        }
//    }
//}
